package controller;

import java.io.IOException;
import java.util.Calendar;
import java.util.Date;
import java.util.StringTokenizer;

import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import models.Domain;
import models.Inlink;

import com.googlecode.objectify.Key;
import com.googlecode.objectify.Objectify;
import com.googlecode.objectify.ObjectifyService;

public class CheckLinkServlet extends HttpServlet {
	public void doGet(HttpServletRequest req, HttpServletResponse resp)
			throws IOException {
		
		resp.setCharacterEncoding("utf-8");
		String path = ((HttpServletRequest)req).getRequestURI();
		StringTokenizer st = new StringTokenizer( path,"/");
        int count = st.countTokens(); 

        if(count != 2)
        {
        	try {
				req.getRequestDispatcher("/404error.jsp").forward(req, resp);
			} catch (Exception e) {
				e.printStackTrace();
			}
        	return;
        }
        
		// skip one token /question/abcd (remove sites)
		st.nextToken();
		String domainName = st.nextToken();
		
		domainName=Utils.fomatDomain(domainName);
		
		if(Utils.checkDomain(domainName))
		{
			resp.setHeader( "Cache-Control", "public, max-age=86400");
			resp.setHeader( "Expires", new Date(Calendar.getInstance().getTime().getTime()+24*60*60*1000).toGMTString());
			try {
				ObjectifyService.register(Domain.class);
			} catch (Exception e) {
				// TODO: handle exception
			}
			Objectify objectify = ObjectifyService.begin();
			try {
				Domain domain = objectify.get(new Key<Domain>(Domain.class, domainName));
	
				req.setAttribute("list", domain);
				
			} catch (Exception e) {
				
			}
			req.setAttribute("domain", domainName);
			
			
			try {
				ObjectifyService.register(Inlink.class);
			} catch (Exception e) {
				// TODO: handle exception
			}
			Objectify objectifys = ObjectifyService.begin();
			try {
				Inlink list = objectifys.get(new Key<Inlink>(Inlink.class, domainName));
	
				req.setAttribute("inlink", list);
				
			} catch (Exception e) {
				
			}
			
			try {
				req.getRequestDispatcher("/check_link.jsp").forward(req, resp);
			} catch (Exception e) {
				e.printStackTrace();
			}
		} else {
			try {
				req.getRequestDispatcher("/404error.jsp").forward(req, resp);
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
	}
}
